package com.leecode.xiehf.page_05;

import com.leecode.Printer;

/**
 * 编写一个函数，输入是一个无符号整数（以二进制串的形式），返回其二进制表达式中数字位数为 '1' 的个数（也被称为汉明重量）。
 * <p>
 *  
 * <p>
 * 提示：
 * <p>
 * 请注意，在某些语言（如 Java）中，没有无符号整数类型。在这种情况下，输入和输出都将被指定为有符号整数类型，并且不应影响您的实现，因为无论整数是有符号的还是无符号的，其内部的二进制表示形式都是相同的。
 * 在 Java 中，编译器使用二进制补码记法来表示有符号整数。因此，在上面的 示例 3 中，输入表示有符号整数 -3。
 *  
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/number-of-1-bits
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class Solution_0208 extends Printer {

    public static void main(String[] args) {
        Solution_0208 solution = new Solution_0208();
        // ["app"],["apple"],["beer"],["add"],["jam"],["rental"],["apps"],["app"],["ad"],["applepie"],["rest"],["jan"],["rent"],["beer"],["jam"],
        // ["apps"],["app"],["ad"],["applepie"],["rest"],["jan"],["rent"],["beer"],["jam"]
        Trie trie = new Trie();
        trie.insert("app");
        trie.insert("apple");
        trie.insert("beer");
        trie.insert("add");
        trie.insert("jam");
        trie.insert("rental");
        trie.insert("apps");
        trie.insert("ad");
        trie.insert("applepie");
        trie.insert("jan");
        trie.insert("beer");
        trie.insert("jam");
        boolean apple = trie.search("apple");
//        System.out.println(trie.startsWith("ab"));
        System.out.println(trie.startsWith("apps"));
        System.out.println(trie.startsWith("app"));
        System.out.println(trie.startsWith("ad"));
        System.out.println(trie.startsWith("applepie"));
        System.out.println(trie.startsWith("rest"));
        System.out.println(trie.startsWith("jan"));
        System.out.println(trie.startsWith("applepie"));
        System.out.println(trie.startsWith("rent"));
        System.out.println(trie.startsWith("jam"));
    }

    static class Trie {

        private String val;

        private Trie left;

        private Trie right;

        /**
         * Initialize your data structure here.
         */
        public Trie() {

        }

        /**
         * Inserts a word into the trie.
         */
        public void insert(String word) {
            if (val == null) {
                val = word;
                return;
            }
            if (val.equals(word)) {
                return;
            }
            if (val.compareTo(word) > 0) {
                if (left == null) {
                    left = new Trie();
                }
                left.insert(word);
            } else {
                if (right == null) {
                    right = new Trie();
                }
                right.insert(word);
            }
        }

        /**
         * Returns if the word is in the trie.
         */
        public boolean search(String word) {
            if (val == null) {
                return false;
            }
            if (val.equals(word)) {
                return true;
            }
            if (val.compareTo(word) > 0) {
                if (left == null) {
                    return false;
                }
                return left.search(word);
            } else {
                if (right == null) {
                    return false;
                }
                return right.search(word);
            }
        }

        /**
         * Returns if there is any word in the trie that starts with the given prefix.
         */
        public boolean startsWith(String prefix) {
            if (val == null) {
                return false;
            }
            if (val.startsWith(prefix)) {
                return true;
            }
            int i = 0, j = 0;
            for (; i < prefix.length() && j < val.length(); i++, j++) {
                if (val.charAt(i) < prefix.charAt(j)) {
                    return right != null && right.startsWith(prefix);
                } else if (val.charAt(i) > prefix.charAt(j)){
                    return left != null && left.startsWith(prefix);
                }
            }
            return right != null && right.startsWith(prefix);
        }
    }

}
